from tools import get_llm
from langchain_core.output_parsers import StrOutputParser
from langchain_core.prompts import ChatPromptTemplate

llm = get_llm()

promptsTemplate = ChatPromptTemplate(
    [
        ('human', '{content}'),
        ('system', '你是一个优秀的AI开发工程师，需要将用户的疑问友好详细的解释清楚')
    ]
)

output_parser = StrOutputParser()

chain = promptsTemplate | llm | output_parser

# 流式输出
result = chain.stream({'content': '我想学习AI请给一份详细的学习文档', 'language': '英文'})
for item in result:
    print(item, end='')
